# 这里是做一个数据的归类 ———— 因为我们图书馆项目当中会出现对于基本数据的添加（作者、出版社等）
# 那么我们就要对不同的信息进行相应的存储，以便于后期的使用和处理

# 而具体作用就是：获取连接 和 关闭连接

# 数据库的工具类

import mysql.connector as mysql_conn


# 获取连接
def get_conn():
    try:
        conn = mysql_conn.connect(host='localhost',
                                  database='library',
                                  user='root',
                                  password='123456')
        return conn
    except mysql_conn.Error:
        print('数据库连接异常')


# 关闭连接 （一定要注意关闭连接，因为十分耗时和占据资源）
def close_conn(conn, cursor):
    try:
        if cursor:
            cursor.close()
        if conn:
            conn.close()
    except mysql_conn.Error:
        print('数据库关闭异常')
    finally:
        cursor.close()
        conn.close()

# print(get_conn())  # 用于测试是否可以数据库的连接
